package com.dy.数组.初级._350_两个数组的交集2;

import java.util.*;

/*
https://leetcode-cn.com/problems/intersection-of-two-arrays-ii/
两个数组的交集 II
给定两个数组，编写一个函数来计算它们的交集。

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明：

输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
进阶:

如果给定的数组已经排好序呢？你将如何优化你的算法？
如果 nums1 的大小比 nums2 小很多，哪种方法更优？
如果 nums2 的元素存储在磁盘上，磁盘内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？
 */
public class Solution {
    public static int[] intersect(int[] nums1, int[] nums2) {
        Map<Integer,Integer> map = new HashMap();
        for(int num1:nums1){
            if(map.containsKey(num1)){
                map.put(num1,map.get(num1)+1);
            }
            else
                map.put(num1,1);
        }
        List<Integer> list = new ArrayList<Integer>();
        for(int num2:nums2 ){
            if(map.containsKey(num2)&&map.get(num2)>0){
                map.put(num2,map.get(num2)-1);
                list.add(num2);
            }
        }
        Iterator iter = list.iterator();
        int rn[]  = new int[list.size()];
        int i=0;
        while (iter.hasNext()){
            rn[i++] =(int) iter.next();
        }
        return rn;

    }
    public static int[] intersect2(int[] nums1, int[] nums2) {
        int n1 = nums1.length;
        int n2 = nums2.length;
        int[] shortNum, longNum;
        if (n1 < n2) {
            shortNum = nums1;
            longNum = nums2;
        } else {
            shortNum = nums2;
            longNum = nums1;
        }
        Map<Integer, Integer> record = new HashMap<>(shortNum.length);
        Integer item;
        for (int moreShortNum : shortNum) {
            item = record.get(moreShortNum);
            if (item == null) {
                record.put(moreShortNum, 1);
            } else {
                record.put(moreShortNum, item + 1);
            }
        }
        List<Integer> res = new ArrayList<>();
        for (int longItem : longNum) {
            item = record.get(longItem);
            if (item != null) {
                if (item != 0) {
                    record.put(longItem, item - 1);
                    res.add(longItem);
                }
            }
        }
        int[] ret = new int[res.size()];
        n1 = 0;
        for (Integer i : res) {
            ret[n1++] = i;
        }
        return ret;
    }
    public static int[] intersect3(int[] nums1, int[] nums2) {
        List<Integer> temp = new ArrayList<>();

        Arrays.sort(nums1);
        Arrays.sort(nums2);
        int i=0;
        int j=0;

        while(i<nums1.length && j<nums2.length){
            if(nums2[j] > nums1[i]){
                i++;
            }else if(nums2[j] < nums1[i]){
                j++;
            }else{
                temp.add(nums1[i]);
                i++;
                j++;

            }
        }

        int[] res = new int[temp.size()];
        for(int m=0;m<res.length;m++){
            res[m]=temp.get(m);
        }
        return res;
    }

    public static int[] intersect4(int[] nums1, int[] nums2) {
        Map<Integer,Integer> map = new HashMap();
        for(int i=0;i<nums1.length;i++){
            int tmp = nums1[i];
            if(map.containsKey(tmp)){
                map.put(tmp,map.get(tmp)+1);
            }else{
                map.put(tmp,1);
            }
        }
        List<Integer> resList = new ArrayList<>();
        for(int i=0;i<nums2.length;i++){
            if(map.containsKey(nums2[i])){
                int count = map.get(nums2[i]);
                if(count==1){
                    map.remove(nums2[i]);
                }else{
                    map.put(nums2[i],count-1);
                }
                resList.add(nums2[i]);
            }
        }
        return resList.stream().mapToInt(Integer::intValue).toArray();

    }

    public static void main(String[] args) {
        Arrays.stream(Solution.intersect4(new int[]{1, 2, 2, 1}, new int[]{2, 2})).forEach(System.out::println);
    }
}
